package main

import (
	"crypto/md5"
	"crypto/sha1"
	"fmt"
	"reflect"

	"github.com/sergi/go-diff/diffmatchpatch"
)

/**
** @ Author       : SmallSuperMan
** @ Date         : 2022-05-04 17:59:54
** @ LastEditTime : 2022-05-09 22:51:06
** @ LastEditors  : SmallSuperMan
** @ FilePath     : \CheckFileMD5\stringTool.go
** @ Description  : 处理字符串的方法汇总
**/

// GetStringHash 返回字符串的hash值字符串
func GetStringHash(inputString string) string {

	h := sha1.New()
	h.Write([]byte(inputString))
	stringHash := h.Sum(nil)

	return string(fmt.Sprintf("%x", stringHash))
}

// GetStringMD5 返回字符串的hash值字符串
func GetStringMD5(inputString string) string {

	h := md5.New()
	h.Write([]byte(inputString))
	stringHash := h.Sum(nil)

	return string(fmt.Sprintf("%x", stringHash))
}

// GetTowStringDiff 获取两个字符不同之处返回信息为 compareString 变成 otherString 需要的操作
func GetTowStringDiff(compareString string, otherString string) []diffmatchpatch.Diff {
	dmp := diffmatchpatch.New()
	diffs := dmp.DiffMain(compareString, otherString, false)
	// log.Printf("diffs--->类型为[%T],[%#v]<---", diffs, diffs)
	return diffs
}

// ArrSliceToMapSet 将切片或者数组转成字典,字典的 key 类型就是输入的元素的值类型
func ArrSliceToMapSet(nullInter interface{}) (interface{}, error) {
	// check param
	if nullInter == nil {
		return nil, fmt.Errorf("转换失败 %#v 类型为 %T 转成 map[interface{}]struct{}", nullInter, nullInter)
	}
	// 获取空接口数据类型
	inType := reflect.TypeOf(nullInter)
	inKind := inType.Kind()

	if !(inKind == reflect.Slice || inKind == reflect.Array) {
		return nil, fmt.Errorf("输入值是 %#v 类型为 %T 既不是切片又不是数组", nullInter, nullInter)
	}

	// 获取空接口传递的值
	tmpV := reflect.ValueOf(nullInter)
	// 获取 key 和 value 的 map 类型 ，例如: MapOf(k, e) 其中k为int, e为string 则返回 map[int]string.
	// 这里 map 的 key 类型是切片或者数组元素类型，值类型是下标的类型也就是int
	myType := reflect.MapOf(inType.Elem(), reflect.TypeOf(1))
	// 创建一个新 map 初始空间 tmpV.Len() 类型是 myType
	myValue := reflect.MakeMapWithSize(myType, tmpV.Len())

	// 遍历依次赋值
	for i := 0; i < tmpV.Len(); i++ {
		myValue.SetMapIndex(tmpV.Index(i), reflect.ValueOf(i))
	}
	// 返回空接口类型 也可以返回 (myValue.Interface()).(map[string]int) map值
	return myValue.Interface(), nil
}

// IsExistsStringInSlice 判断字符串是否在某个切片中
func IsExistsStringInSlice(element string, inSlice []string) bool {
	inSliceToMap, _ := ArrSliceToMapSet(inSlice)
	_, ok := (inSliceToMap.(map[string]int))[element]
	return ok
}

// IsGBKK()是通过双字节是否落在gbk的编码范围内实现的，
// 而utf-8编码格式的每个字节都是落在gbk的编码范围内，
// 所以只有先调用isUtf8()先判断不是utf-8编码，再调用isGBK()才有意义
// IsGBK GBK编码格式判断
func IsGBK(data []byte) bool {
	length := len(data)
	var i int = 0
	for i < length {
		if data[i] <= 0x7f {
			//编码0~127,只有一个字节的编码，兼容ASCII码
			i++
			continue
		} else {
			//大于127的使用双字节编码，落在gbk编码范围内的字符
			if data[i] >= 0x81 &&
				data[i] <= 0xfe &&
				data[i+1] >= 0x40 &&
				data[i+1] <= 0xfe &&
				data[i+1] != 0xf7 {
				i += 2
				continue
			} else {
				return false
			}
		}
	}
	return true
}

// UTF-8编码格式的判断
func PreNum(data byte) int {
	var mask byte = 0x80
	var num int = 0
	//8bit中首个0bit前有多少个1bits
	for i := 0; i < 8; i++ {
		if (data & mask) == mask {
			num++
			mask = mask >> 1
		} else {
			break
		}
	}
	return num
}

// IsUtf8 UTF-8编码格式的判断
func IsUtf8(data []byte) bool {
	i := 0
	for i < len(data) {
		if (data[i] & 0x80) == 0x00 {
			// 0XXX_XXXX
			i++
			continue
		} else if num := PreNum(data[i]); num > 2 {
			// 110X_XXXX 10XX_XXXX
			// 1110_XXXX 10XX_XXXX 10XX_XXXX
			// 1111_0XXX 10XX_XXXX 10XX_XXXX 10XX_XXXX
			// 1111_10XX 10XX_XXXX 10XX_XXXX 10XX_XXXX 10XX_XXXX
			// 1111_110X 10XX_XXXX 10XX_XXXX 10XX_XXXX 10XX_XXXX 10XX_XXXX
			// preNUm() 返回首个字节的8个bits中首个0bit前面1bit的个数，该数量也是该字符所使用的字节数
			i++
			for j := 0; j < num-1; j++ {
				//判断后面的 num - 1 个字节是不是都是10开头
				if (data[i] & 0xc0) != 0x80 {
					return false
				}
				i++
			}
		} else {
			//其他情况说明不是utf-8
			return false
		}
	}
	return true
}
